package com.woniuxy.util;

public class ImageFormatDetector {
    public enum ImageFormat {
        PNG, JPEG, BMP, UNKNOWN
    }

    public static ImageFormat detect(byte[] imageData) {
        if (imageData == null || imageData.length < 8) {
            return ImageFormat.UNKNOWN;
        }

        // PNG检查
        if (imageData[0] == (byte) 0x89 &&
                new String(imageData, 1, 3).equals("PNG")) {
            return ImageFormat.PNG;
        }

        // JPEG检查
        if (imageData[0] == (byte) 0xFF &&
                imageData[1] == (byte) 0xD8) {
            return ImageFormat.JPEG;
        }

        // BMP检查
        if (new String(imageData, 0, 2).equals("BM")) {
            return ImageFormat.BMP;
        }

        return ImageFormat.UNKNOWN;
    }
}